from .dalchemy_task import DalchemyTask
import os
import logging
import importlib
'''
待添加：
1.一般的nlp任务，包括翻译、问答、摘要等
2.扩充数据的流程，参考alpaca。每次从seed和上两轮生成的sample中采样 [组合多个task，是否需要pipeline类？]
3.多轮会话，在llm修改
'''

def import_models(infer_dir, namespace):
    for file in os.listdir(infer_dir):
        path = os.path.join(infer_dir, file)
        if (
            not file.startswith("_")
            and not file.startswith(".")
            and (file.endswith(".py") or os.path.isdir(path))
        ):
            infer_name = file[: file.find(".py")] if file.endswith(".py") else file
            importlib.import_module(namespace + "." + infer_name)


# automatically import any Python files in the infer/ directory
infer_dir = os.path.dirname(__file__)
import_models(infer_dir, "dalchemy.tasks")